import torch

# 检查GPU是否可用
if torch.cuda.is_available():
    print("GPU可用")
    # 获取GPU数量
    print("GPU数量:", torch.cuda.device_count())
    # 获取第一个GPU的名称
    print("GPU名称:", torch.cuda.get_device_name(0))
    # 获取当前GPU的索引
    print("当前GPU索引:", torch.cuda.current_device())
else:
    print("GPU不可用")